Create new grouped conditional variable in R